package com.reformer.commons.utils;

import freemarker.cache.StringTemplateLoader;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Locale;
import java.util.Map;

/**
 * freemarker模板工具类
 *
 * @author sdc
 * @since 2.0
 */
public class FreemarkerUtil {

    /**
     * 打印到控制台(测试用)
     *
     * @param ftlName
     * @deprecated
     */
    public static void print(String ftlName, Map<String, Object> root, String ftlPath) throws Exception {
        try {
            Template temp = getTemplate(ftlName);        //通过Template可以将模板文件输出到相应的流
            temp.process(root, new PrintWriter(System.out));
        } catch (TemplateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 输出到输出到文件
     *
     * @param ftlName  ftl文件名
     * @param root     传入的map
     * @param outFile  输出后的文件全部路径
     * @param filePath 输出前的文件上部路径
     * @deprecated
     */
    public static void printFile(String ftlName, Map<String, Object> root, String outFile, String filePath, String ftlPath) throws Exception {
        try {
            File file = new File(PathUtil.getClasspath() + filePath + outFile);
            if (!file.getParentFile().exists()) {                //判断有没有父路径，就是判断文件整个路径是否存在
                file.getParentFile().mkdirs();                //不存在就全部创建
            }
            Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));
            Template template = getTemplate(ftlName);
            template.process(root, out);                    //模版输出
            out.flush();
            out.close();
        } catch (TemplateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 加载classpath/ftl目录下文件为ftlName的模版
     *
     * @param ftlName
     */
    public static Template getTemplate(String ftlName) {
        try {
            //通过Freemaker的Configuration读取相应的ftl
            Configuration cfg = new Configuration();
            cfg.setEncoding(Locale.CHINA, "utf-8");
            //设定模板文件目录
            cfg.setDirectoryForTemplateLoading(new File(PathUtil.getClasspath(), "ftl"));
            return cfg.getTemplate(ftlName);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 输出模板
     *
     * @param ftlName   模板文件
     * @param out       输出流
     * @param dataModel 数据模型
     * @throws IOException
     * @throws TemplateException
     */
    public static void output(String ftlName, Writer out, Map<String, Object> dataModel) throws TemplateException, IOException {
        Template template = getTemplate(ftlName);
        template.process(dataModel, out);
    }

    /**
     * 渲染字符串模板
     *
     * @param templateSource 模板字符串
     * @param dataModel      数据模型
     * @return 渲染后的字符串
     * @throws TemplateException
     * @throws IOException
     */
    public static String render(String templateSource, Map<String, Object> dataModel) throws TemplateException, IOException {
        Configuration cfg = new Configuration();
        cfg.setEncoding(Locale.CHINA, "utf-8");
        StringTemplateLoader stringLoader = new StringTemplateLoader();
        stringLoader.putTemplate("myTemplate", templateSource);
        cfg.setTemplateLoader(stringLoader);
        Template template = cfg.getTemplate("myTemplate", "utf-8");
        Writer out = new StringWriter(2048);
        template.process(dataModel, out);
        return out.toString();
    }
}
